from config import BASE_HOST
class RechargeApi:
    def __init__(self,ses):
    # 接收session对象的变量
        self.ses = ses
    # 获取充值验证码接口
    def get_recharge_verify_code(self, r):
        # 定义一个变量，用来接收接口请求得url
        url = BASE_HOST + "/common/public/verifycode/{}".format(r)
        # 实现接口请求，获取响应数据
        resp = self.ses.get(url=url)
        print("获取充值验证码接口返回的响应数据：{}".format(resp))
        # 返回结果
        return resp
    # 充值接口
    def recharge(self, amount_v, imgCode_v, paymentType_v="chinapnrTrust", formStr_v="reForm"):
        # 定义一个变量，用来接收接口请求得url
        url = BASE_HOST + "/trust/trust/recharge"
        # 定义一个字典变量，用来接收请求体数据
        form_dict = {"paymentType":paymentType_v,"amount":amount_v,"formStr":formStr_v,"valicode":imgCode_v}
        # 实现接口请求，获取响应数据
        resp = self.ses.post(url=url,data=form_dict)
        print("获取充值接口返回的响应数据：{}".format(resp))
        # 返回结果
        return resp

    # 第三方充值接口
    def third_recharge(self, url, form_dict):
        # 实现接口请求，获取响应结果
        resp = self.ses.post(url=url,data=form_dict)
        print("获取充值接口返回的响应数据：{}".format(resp))
        # 封装方法的返回
        return resp